#!/usr/bin/env python3
"""HIV/AIDS Analysis Script
Author: Juan Moises de la Serna
"""
import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('hiv_extended.csv')
print(df.groupby('country')['prevalence_pct'].describe())

for country in df['country'].unique():
    sub = df[df['country'] == country]
    plt.plot(sub['year'], sub['art_coverage_pct'], marker='o', label=country)
plt.title('ART Coverage Trends')
plt.xlabel('Year')
plt.ylabel('ART Coverage (%)')
plt.legend()
plt.tight_layout()
plt.savefig('art_coverage.png', dpi=150)
